import random
import torch
# rom d2l import torch as d2l